Node / ReticulumProjects / MeshChatX.git / files / tests / backend / test_self_check_cli.py

Displaying Raw • Download

T8b949e# SPDX-License-Identifier: 0BSD

Tff7b72import T7ee787shutil

Tff7b72import T7ee787tempfile

Tff7b72from T7ee787unittestT7ee787.T7ee787mock Tff7b72import Te6edf3MagicMockTb4b4b4, Te6edf3patch

Tff7b72import T7ee787pytest

Tff7b72import T7ee787RNS

Tff7b72from T7ee787meshchatxT7ee787.T7ee787meshchat Tff7b72import Te6edf3ReticulumMeshChatTb4b4b4, Te6edf3main

Tf0883e@pytestTff7b72.Td2a8fffixture

Tff7b72def Td2a8fftemp_dirTb4b4b4(Tb4b4b4)Tb4b4b4:

Te6edf3dir_path Tff7b72= Te6edf3tempfileTff7b72.Td2a8ffmkdtempTb4b4b4(Tb4b4b4)

Tff7b72yield Te6edf3dir_path

Te6edf3shutilTff7b72.Td2a8ffrmtreeTb4b4b4(Te6edf3dir_pathTb4b4b4)

Tf0883e@pytestTff7b72.Td2a8fffixture

Tff7b72def Td2a8ffmock_rnsTb4b4b4(Tb4b4b4)Tb4b4b4:

T8b949e# Save the real identity class to use as base for our mock class

Te6edf3real_identity_class Tff7b72= Te6edf3RNSTff7b72.Td2a8ffIdentity

Tff7b72class T56d364MockIdentityClassTb4b4b4(Te6edf3real_identity_classTb4b4b4)Tb4b4b4:

Tff7b72def Tff7b72__init__Tb4b4b4(Tff7b72selfTb4b4b4, Tff7b72*Te6edf3argsTb4b4b4, Tff7b72*Tff7b72*Te6edf3kwargsTb4b4b4)Tb4b4b4:

Tff7b72selfTff7b72.Td2a8ffhash Tff7b72= Ta5d6ffbTa5d6ff"Ta5d6fftest_hash_32_bytes_long_01234567Ta5d6ff"

Tff7b72selfTff7b72.Td2a8ffhexhash Tff7b72= Tff7b72selfTff7b72.Td2a8ffhashTff7b72.Td2a8ffhexTb4b4b4(Tb4b4b4)

Tff7b72def Td2a8ffget_private_keyTb4b4b4(Tff7b72selfTb4b4b4)Tb4b4b4:

Tff7b72return Ta5d6ffbTa5d6ff"Ta5d6fftest_private_keyTa5d6ff"

Tff7b72def Td2a8ffloadTb4b4b4(Tff7b72selfTb4b4b4, Tff7b72*Te6edf3argsTb4b4b4, Tff7b72*Tff7b72*Te6edf3kwargsTb4b4b4)Tb4b4b4:

Tff7b72pass

Tff7b72def Td2a8ffload_private_keyTb4b4b4(Tff7b72selfTb4b4b4, Tff7b72*Te6edf3argsTb4b4b4, Tff7b72*Tff7b72*Te6edf3kwargsTb4b4b4)Tb4b4b4:

Tff7b72pass

Tff7b72with Tb4b4b4(

Te6edf3patchTb4b4b4(Ta5d6ff"Ta5d6ffRNS.ReticulumTa5d6ff"Tb4b4b4) Tff7b72as Te6edf3mock_reticulumTb4b4b4,

Te6edf3patchTb4b4b4(Ta5d6ff"Ta5d6ffRNS.TransportTa5d6ff"Tb4b4b4) Tff7b72as Te6edf3mock_transportTb4b4b4,

Te6edf3patchTb4b4b4(Ta5d6ff"Ta5d6ffRNS.IdentityTa5d6ff"Tb4b4b4, Te6edf3MockIdentityClassTb4b4b4)Tb4b4b4,

Te6edf3patchTb4b4b4(Ta5d6ff"Ta5d6ffthreading.ThreadTa5d6ff"Tb4b4b4)Tb4b4b4,

Te6edf3patchTb4b4b4(Ta5d6ff"Ta5d6ffLXMF.LXMRouterTa5d6ff"Tb4b4b4)Tb4b4b4,

Te6edf3patchTff7b72.Td2a8ffobjectTb4b4b4(Te6edf3ReticulumMeshChatTb4b4b4, Ta5d6ff"Ta5d6ffannounce_loopTa5d6ff"Tb4b4b4, Te6edf3return_valueTff7b72=Tff7b72NoneTb4b4b4)Tb4b4b4,

Te6edf3patchTff7b72.Td2a8ffobjectTb4b4b4(

Te6edf3ReticulumMeshChatTb4b4b4,

Ta5d6ff"Ta5d6ffannounce_sync_propagation_nodesTa5d6ff"Tb4b4b4,

Te6edf3return_valueTff7b72=Tff7b72NoneTb4b4b4,

Tb4b4b4)Tb4b4b4,

Te6edf3patchTff7b72.Td2a8ffobjectTb4b4b4(Te6edf3ReticulumMeshChatTb4b4b4, Ta5d6ff"Ta5d6ffcrawler_loopTa5d6ff"Tb4b4b4, Te6edf3return_valueTff7b72=Tff7b72NoneTb4b4b4)Tb4b4b4,

Te6edf3patchTff7b72.Td2a8ffobjectTb4b4b4(Te6edf3ReticulumMeshChatTb4b4b4, Ta5d6ff"Ta5d6ffauto_backup_loopTa5d6ff"Tb4b4b4, Te6edf3return_valueTff7b72=Tff7b72NoneTb4b4b4)Tb4b4b4,

Te6edf3patchTff7b72.Td2a8ffobjectTb4b4b4(

Te6edf3ReticulumMeshChatTb4b4b4,

Ta5d6ff"Ta5d6ffsend_config_to_websocket_clientsTa5d6ff"Tb4b4b4,

Te6edf3return_valueTff7b72=Tff7b72NoneTb4b4b4,

Tb4b4b4)Tb4b4b4,

Tb4b4b4)Tb4b4b4:

Te6edf3mock_id_instance Tff7b72= Te6edf3MockIdentityClassTb4b4b4(Tb4b4b4)

Tff7b72with Tb4b4b4(

Te6edf3patchTff7b72.Td2a8ffobjectTb4b4b4(Te6edf3MockIdentityClassTb4b4b4, Ta5d6ff"Ta5d6fffrom_fileTa5d6ff"Tb4b4b4, Te6edf3return_valueTff7b72=Te6edf3mock_id_instanceTb4b4b4)Tb4b4b4,

Te6edf3patchTff7b72.Td2a8ffobjectTb4b4b4(Te6edf3MockIdentityClassTb4b4b4, Ta5d6ff"Ta5d6ffrecallTa5d6ff"Tb4b4b4, Te6edf3return_valueTff7b72=Te6edf3mock_id_instanceTb4b4b4)Tb4b4b4,

Te6edf3patchTff7b72.Td2a8ffobjectTb4b4b4(

Te6edf3MockIdentityClassTb4b4b4,

Ta5d6ff"Ta5d6fffrom_bytesTa5d6ff"Tb4b4b4,

Te6edf3return_valueTff7b72=Te6edf3mock_id_instanceTb4b4b4,

Tb4b4b4)Tb4b4b4,

Tb4b4b4)Tb4b4b4:

Tff7b72yield Tb4b4b4{

Ta5d6ff"Ta5d6ffReticulumTa5d6ff"Tb4b4b4: Te6edf3mock_reticulumTb4b4b4,

Ta5d6ff"Ta5d6ffTransportTa5d6ff"Tb4b4b4: Te6edf3mock_transportTb4b4b4,

Ta5d6ff"Ta5d6ffIdentityTa5d6ff"Tb4b4b4: Te6edf3MockIdentityClassTb4b4b4,

Ta5d6ff"Ta5d6ffid_instanceTa5d6ff"Tb4b4b4: Te6edf3mock_id_instanceTb4b4b4,

Tb4b4b4}

Tff7b72def Td2a8ff_ok_resultsTb4b4b4(Tb4b4b4)Tb4b4b4:

Tff7b72from T7ee787meshchatxT7ee787.T7ee787srcT7ee787.T7ee787backendT7ee787.T7ee787self_check Tff7b72import Te6edf3SELF_CHECK_LABELS

Tff7b72return Tb4b4b4{Te6edf3keyTb4b4b4: Tb4b4b4{Ta5d6ff"Ta5d6ffstatusTa5d6ff"Tb4b4b4: Ta5d6ff"Ta5d6ffokTa5d6ff"Tb4b4b4, Ta5d6ff"Ta5d6ffreasonTa5d6ff"Tb4b4b4: Ta5d6ff"Ta5d6ff"Tb4b4b4} Tff7b72for Te6edf3key Tff7b72in Te6edf3SELF_CHECK_LABELSTb4b4b4}

Tff7b72def Td2a8fftest_self_check_cli_successTb4b4b4(Te6edf3mock_rnsTb4b4b4, Te6edf3temp_dirTb4b4b4)Tb4b4b4:

T8b949e"""Test that self-check CLI argument prints results and exits with 0 on success."""

Te6edf3mock_results Tff7b72= Te6edf3_ok_resultsTb4b4b4(Tb4b4b4)

Tff7b72with Tb4b4b4(

Te6edf3patchTb4b4b4(Ta5d6ff"Ta5d6ffmeshchatx.meshchat.ReticulumMeshChatTa5d6ff"Tb4b4b4) Tff7b72as Te6edf3mock_app_classTb4b4b4,

Te6edf3patchTb4b4b4(Ta5d6ff"Ta5d6ffmeshchatx.src.backend.identity_context.DatabaseTa5d6ff"Tb4b4b4)Tb4b4b4,

Te6edf3patchTb4b4b4(Ta5d6ff"Ta5d6ffmeshchatx.src.backend.identity_context.ConfigManagerTa5d6ff"Tb4b4b4)Tb4b4b4,

Te6edf3patchTb4b4b4(Ta5d6ff"Ta5d6ffaiohttp.web.run_appTa5d6ff"Tb4b4b4)Tb4b4b4,

Te6edf3patchTb4b4b4(Ta5d6ff"Ta5d6ffsys.argvTa5d6ff"Tb4b4b4, Tb4b4b4[Ta5d6ff"Ta5d6ffmeshchat.pyTa5d6ff"Tb4b4b4, Ta5d6ff"Ta5d6ff--storage-dirTa5d6ff"Tb4b4b4, Te6edf3temp_dirTb4b4b4, Ta5d6ff"Ta5d6ff--self-checkTa5d6ff"Tb4b4b4]Tb4b4b4)Tb4b4b4,

Tb4b4b4)Tb4b4b4:

Te6edf3mock_app_instance Tff7b72= Te6edf3mock_app_classTff7b72.Td2a8ffreturn_value

Te6edf3mock_app_instanceTff7b72.Td2a8ffrun_self_test Tff7b72= Te6edf3MagicMockTb4b4b4(Te6edf3return_valueTff7b72=Te6edf3mock_resultsTb4b4b4)

Tff7b72with Te6edf3pytestTff7b72.Td2a8ffraisesTb4b4b4(Tf85149SystemExitTb4b4b4) Tff7b72as Te6edf3excinfoTb4b4b4:

Te6edf3mainTb4b4b4(Tb4b4b4)

Tff7b72assert Te6edf3excinfoTff7b72.Td2a8ffvalueTff7b72.Td2a8ffcode Tff7b72== T79c0ff0

Te6edf3mock_app_instanceTff7b72.Td2a8ffrun_self_testTff7b72.Td2a8ffassert_called_onceTb4b4b4(Tb4b4b4)

Tff7b72def Td2a8fftest_self_check_cli_failureTb4b4b4(Te6edf3mock_rnsTb4b4b4, Te6edf3temp_dirTb4b4b4)Tb4b4b4:

T8b949e"""Test that self-check CLI argument prints results and exits with 1 on failure."""

Te6edf3mock_results Tff7b72= Te6edf3_ok_resultsTb4b4b4(Tb4b4b4)

Te6edf3mock_resultsTb4b4b4[Ta5d6ff"Ta5d6ffstack_upTa5d6ff"Tb4b4b4] Tff7b72= Tb4b4b4{

Ta5d6ff"Ta5d6ffstatusTa5d6ff"Tb4b4b4: Ta5d6ff"Ta5d6fffailedTa5d6ff"Tb4b4b4,

Ta5d6ff"Ta5d6ffreasonTa5d6ff"Tb4b4b4: Ta5d6ff"Ta5d6ffReticulum stack is not initializedTa5d6ff"Tb4b4b4,

Tb4b4b4}

Te6edf3mock_resultsTb4b4b4[Ta5d6ff"Ta5d6ffdb_goodTa5d6ff"Tb4b4b4] Tff7b72= Tb4b4b4{

Ta5d6ff"Ta5d6ffstatusTa5d6ff"Tb4b4b4: Ta5d6ff"Ta5d6fffailedTa5d6ff"Tb4b4b4,

Ta5d6ff"Ta5d6ffreasonTa5d6ff"Tb4b4b4: Ta5d6ff"Ta5d6ffDatabase check failedTa5d6ff"Tb4b4b4,

Tb4b4b4}

Tff7b72with Tb4b4b4(

Te6edf3patchTb4b4b4(Ta5d6ff"Ta5d6ffmeshchatx.meshchat.ReticulumMeshChatTa5d6ff"Tb4b4b4) Tff7b72as Te6edf3mock_app_classTb4b4b4,

Te6edf3patchTb4b4b4(Ta5d6ff"Ta5d6ffmeshchatx.src.backend.identity_context.DatabaseTa5d6ff"Tb4b4b4)Tb4b4b4,

Te6edf3patchTb4b4b4(Ta5d6ff"Ta5d6ffmeshchatx.src.backend.identity_context.ConfigManagerTa5d6ff"Tb4b4b4)Tb4b4b4,

Te6edf3patchTb4b4b4(Ta5d6ff"Ta5d6ffaiohttp.web.run_appTa5d6ff"Tb4b4b4)Tb4b4b4,

Te6edf3patchTb4b4b4(Ta5d6ff"Ta5d6ffsys.argvTa5d6ff"Tb4b4b4, Tb4b4b4[Ta5d6ff"Ta5d6ffmeshchat.pyTa5d6ff"Tb4b4b4, Ta5d6ff"Ta5d6ff--storage-dirTa5d6ff"Tb4b4b4, Te6edf3temp_dirTb4b4b4, Ta5d6ff"Ta5d6ff--self-checkTa5d6ff"Tb4b4b4]Tb4b4b4)Tb4b4b4,

Tb4b4b4)Tb4b4b4:

Te6edf3mock_app_instance Tff7b72= Te6edf3mock_app_classTff7b72.Td2a8ffreturn_value

Te6edf3mock_app_instanceTff7b72.Td2a8ffrun_self_test Tff7b72= Te6edf3MagicMockTb4b4b4(Te6edf3return_valueTff7b72=Te6edf3mock_resultsTb4b4b4)

Tff7b72with Te6edf3pytestTff7b72.Td2a8ffraisesTb4b4b4(Tf85149SystemExitTb4b4b4) Tff7b72as Te6edf3excinfoTb4b4b4:

Te6edf3mainTb4b4b4(Tb4b4b4)

Tff7b72assert Te6edf3excinfoTff7b72.Td2a8ffvalueTff7b72.Td2a8ffcode Tff7b72== T79c0ff1

Te6edf3mock_app_instanceTff7b72.Td2a8ffrun_self_testTff7b72.Td2a8ffassert_called_onceTb4b4b4(Tb4b4b4)

Tff7b72def Td2a8fftest_self_check_env_var_successTb4b4b4(Te6edf3mock_rnsTb4b4b4, Te6edf3temp_dirTb4b4b4)Tb4b4b4:

T8b949e"""Test that MESHCHAT_SELF_CHECK env var triggers self-check and exits with 0 on success."""

Te6edf3mock_results Tff7b72= Te6edf3_ok_resultsTb4b4b4(Tb4b4b4)

Te6edf3env Tff7b72= Tb4b4b4{

Ta5d6ff"Ta5d6ffMESHCHAT_SELF_CHECKTa5d6ff"Tb4b4b4: Ta5d6ff"Ta5d6fftrueTa5d6ff"Tb4b4b4,

Ta5d6ff"Ta5d6ffMESHCHAT_STORAGE_DIRTa5d6ff"Tb4b4b4: Te6edf3temp_dirTb4b4b4,

Tb4b4b4}

Tff7b72with Tb4b4b4(

Te6edf3patchTb4b4b4(Ta5d6ff"Ta5d6ffmeshchatx.meshchat.ReticulumMeshChatTa5d6ff"Tb4b4b4) Tff7b72as Te6edf3mock_app_classTb4b4b4,

Te6edf3patchTb4b4b4(Ta5d6ff"Ta5d6ffmeshchatx.src.backend.identity_context.DatabaseTa5d6ff"Tb4b4b4)Tb4b4b4,

Te6edf3patchTb4b4b4(Ta5d6ff"Ta5d6ffmeshchatx.src.backend.identity_context.ConfigManagerTa5d6ff"Tb4b4b4)Tb4b4b4,

Te6edf3patchTb4b4b4(Ta5d6ff"Ta5d6ffaiohttp.web.run_appTa5d6ff"Tb4b4b4)Tb4b4b4,

Te6edf3patchTff7b72.Td2a8ffdictTb4b4b4(Ta5d6ff"Ta5d6ffos.environTa5d6ff"Tb4b4b4, Te6edf3envTb4b4b4)Tb4b4b4,

Te6edf3patchTb4b4b4(Ta5d6ff"Ta5d6ffsys.argvTa5d6ff"Tb4b4b4, Tb4b4b4[Ta5d6ff"Ta5d6ffmeshchat.pyTa5d6ff"Tb4b4b4]Tb4b4b4)Tb4b4b4,

Tb4b4b4)Tb4b4b4:

Te6edf3mock_app_instance Tff7b72= Te6edf3mock_app_classTff7b72.Td2a8ffreturn_value

Te6edf3mock_app_instanceTff7b72.Td2a8ffrun_self_test Tff7b72= Te6edf3MagicMockTb4b4b4(Te6edf3return_valueTff7b72=Te6edf3mock_resultsTb4b4b4)

Tff7b72with Te6edf3pytestTff7b72.Td2a8ffraisesTb4b4b4(Tf85149SystemExitTb4b4b4) Tff7b72as Te6edf3excinfoTb4b4b4:

Te6edf3mainTb4b4b4(Tb4b4b4)

Tff7b72assert Te6edf3excinfoTff7b72.Td2a8ffvalueTff7b72.Td2a8ffcode Tff7b72== T79c0ff0

Te6edf3mock_app_instanceTff7b72.Td2a8ffrun_self_testTff7b72.Td2a8ffassert_called_onceTb4b4b4(Tb4b4b4)